<template>
  <el-dialog :visible.sync="dialogVisible" width="975px" center custom-class="custom-dialog" :show-close="false"
    class="center-dialog">
    <div style="margin: 0 18px; text-align: left; padding: 10px; border-radius: 10px;">
      <div style="font-size: 30px; color: #3d4566; margin-top: -10px; margin-bottom: 10px; text-align: center;">
        修改模型
      </div>

      <button class="custom-close-btn" @click="dialogVisible = false">
        ×
      </button>

      <div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 10px;">
        <div style="font-size: 20px; font-weight: bold; color: #3d4566;">模型信息</div>
        <div style="display: flex; align-items: center; gap: 20px;">
          <div style="display: flex; align-items: center;">
            <span style="margin-right: 8px;">是否启用</span>
            <el-switch v-model="form.isEnabled" :active-value="1" :inactive-value="0" class="custom-switch"></el-switch>
          </div>
          <div style="display: flex; align-items: center;">
            <span style="margin-right: 8px;">设为默认</span>
            <el-switch v-model="form.isDefault" :active-value="1" :inactive-value="0" class="custom-switch"></el-switch>
          </div>
        </div>
      </div>

      <div style="height: 2px; background: #e9e9e9; margin-bottom: 22px;"></div>

      <el-form :model="form" ref="form" label-width="100px" label-position="left" class="custom-form">
        <div style="display: flex; gap: 20px; margin-bottom: 0;">
          <el-form-item label="模型名称" prop="name" style="flex: 1;">
            <el-input v-model="form.modelName" placeholder="请输入模型名称" class="custom-input-bg"></el-input>
          </el-form-item>
          <el-form-item label="模型编码" prop="code" style="flex: 1;">
            <el-input v-model="form.modelCode" placeholder="请输入模型编码" class="custom-input-bg"></el-input>
          </el-form-item>
        </div>

        <div style="display: flex; gap: 20px; margin-bottom: 0;">
          <el-form-item label="供应器" prop="supplier" style="flex: 1;">
            <el-select v-model="form.configJson.type" placeholder="请选择" class="custom-select custom-input-bg"
              style="width: 100%;" @focus="loadProviders" filterable>
              <el-option v-for="item in providers" :key="item.value" :label="item.label" :value="item.value" />
            </el-select>
          </el-form-item>
          <el-form-item label="排序号" prop="sort" style="flex: 1;">
            <el-input v-model="form.sort" placeholder="请输入排序号" class="custom-input-bg"></el-input>
          </el-form-item>
        </div>

        <el-form-item label="文档地址" prop="docUrl" style="margin-bottom: 27px;">
          <el-input v-model="form.docLink" placeholder="请输入文档地址" class="custom-input-bg"></el-input>
        </el-form-item>

        <el-form-item label="备注" prop="remark" class="prop-remark">
          <el-input v-model="form.remark" type="textarea" :rows="3" placeholder="请输入模型备注"
            class="custom-input-bg"></el-input>
        </el-form-item>
      </el-form>

      <div style="font-size: 20px; font-weight: bold; color: #3d4566; margin-bottom: 15px;">调用信息</div>
      <div style="height: 2px; background: #e9e9e9; margin-bottom: 22px;"></div>

      <el-form :model="form.configJson" ref="callInfoForm" label-width="100px" label-position="left" class="custom-form">
        <template v-for="(row, rowIndex) in chunkedCallInfoFields">
          <div :key="rowIndex" style="display: flex; gap: 20px; margin-bottom: 0;">
            <el-form-item v-for="field in row"
              :key="field.prop"
              :label="field.label"
              :prop="field.prop"
              style="flex: 1;">
              <el-input
                v-model="form.configJson[field.prop]"
                :placeholder="field.placeholder"
                :type="field.type || 'text'"
                class="custom-input-bg"
                :show-password="field.type === 'password'">
              </el-input>
            </el-form-item>
          </div>
        </template>
      </el-form>
    </div>

    <div style="display: flex;justify-content: center;">
      <el-button type="primary" @click="handleSave" class="save-btn">
        保存
      </el-button>
    </div>
  </el-dialog>
</template>

<script>
import Api from '@/apis/api';

const DEFAULT_CONFIG_JSON = {
  type: "",
  base_url: "",
  model_name: "",
  api_key: "",
  raw: {},
  config: {
    keyComparator: {},
    ignoreError: false,
    ignoreCase: false,
    dateFormat: "",
    ignoreNullValue: false,
    transientSupport: false,
    stripTrailingZeros: false,
    checkDuplicate: false,
    order: false
  },
  empty: false
};

export default {
  name: "ModelEditDialog",
  props: {
    visible: { type: Boolean, default: false },
    modelData: {
      type: Object,
      default: () => ({}),
      validator: value => typeof value === 'object' && !Array.isArray(value)
    },
    modelType: { type: String, required: true }
  },
  data() {
    return {
      dialogVisible: this.visible,
      providers: [],
      providersLoaded: false,
      form: {
        id: "",
        modelType: "",
        modelCode: "",
        modelName: "",
        isDefault: false,
        isEnabled: false,
        docLink: "",
        remark: "",
        sort: 0,
        configJson: JSON.parse(JSON.stringify(DEFAULT_CONFIG_JSON))
      }
    };
  },
  computed: {
    callInfoFields() {
      const fieldsMap = {
        llm: [
            { label: '模型名称', prop: 'model_name', placeholder: '请输入model_name' },
            { label: '接口地址', prop: 'base_url', placeholder: '请输入base_url' },
            { label: '秘钥信息', prop: 'api_key', placeholder: '请输入api_key', type: 'password' }
        ],
        vad: [
            { label: '模型名称', prop: 'model_name', placeholder: '请输入model_name' },
            { label: '模型目录', prop: 'model_dir', placeholder: '请输入model_dir' },
            { label: '阈值', prop: 'threshold', placeholder: '请输入threshold' },
            { label: '静音时长', prop: 'min_silence_duration_ms', placeholder: '请输入min_silence_duration_ms' },
            { label: '接口地址', prop: 'base_url', placeholder: '请输入base_url' },
            { label: '秘钥信息', prop: 'api_key', placeholder: '请输入api_key', type: 'password' }
        ],
        asr: [
            { label: '模型名称', prop: 'model_name', placeholder: '请输入model_name' },
            { label: '集群', prop: 'cluster', placeholder: '请输入cluster' },
            { label: '接口地址', prop: 'base_url', placeholder: '请输入base_url' },
            { label: '秘钥信息', prop: 'api_key', placeholder: '请输入api_key', type: 'password' }
        ],
        intent: [
            { label: '模型名称', prop: 'model_name', placeholder: '请输入model_name' },
            { label: 'LLM模型', prop: 'llm', placeholder: '请输入llm' },
            { label: '接口地址', prop: 'base_url', placeholder: '请输入base_url' },
            { label: '秘钥信息', prop: 'api_key', placeholder: '请输入api_key', type: 'password' }
        ],
        tts: [
            { label: '模型名称', prop: 'model_name', placeholder: '请输入model_name' },
            { label: '语音ID', prop: 'voice_id', placeholder: '请输入voice_id' },
            { label: '接口地址', prop: 'base_url', placeholder: '请输入base_url' },
            { label: '秘钥信息', prop: 'api_key', placeholder: '请输入api_key', type: 'password' }
        ],
        memory: [
            { label: '模型名称', prop: 'model_name', placeholder: '请输入model_name' },
            { label: '接口地址', prop: 'base_url', placeholder: '请输入base_url' },
            { label: '秘钥信息', prop: 'api_key', placeholder: '请输入api_key', type: 'password' }
        ]
      };

      return fieldsMap[this.modelType] || [];
    },
    chunkedCallInfoFields() {
      const chunkSize = 2;
      const result = [];
      for (let i = 0; i < this.callInfoFields.length; i += chunkSize) {
        result.push(this.callInfoFields.slice(i, i + chunkSize));
      }
      return result;
    },
  },
  watch: {
    modelType() {
      this.resetProviders()
    },
    dialogVisible(val) {
      this.$emit('update:visible', val);
      if (!val) {
        this.resetForm();
      } else if (val && this.modelData.id) {
        this.loadModelData();
      }
    },
    visible(val) {
      this.dialogVisible = val;
    }
  },
  methods: {
    resetForm() {
      this.form = {
        id: "",
        modelType: "",
        modelCode: "",
        modelName: "",
        isDefault: false,
        isEnabled: false,
        docLink: "",
        remark: "",
        sort: 0,
        configJson: JSON.parse(JSON.stringify(DEFAULT_CONFIG_JSON))
      };
    },
    resetProviders() {
      this.providers = [];
      this.providersLoaded = false;
    },
    loadModelData() {
      if (this.modelData.id) {
        Api.model.getModelConfig(this.modelData.id, ({ data }) => {
          if (data.code === 0 && data.data) {
            const model = data.data;

            let configJson = model.configJson || {};
            if (typeof configJson !== 'object' || Array.isArray(configJson)) {
              console.warn('Invalid configJson format, using default');
              configJson = {};
            }

            this.callInfoFields.forEach(field => {
              if (!configJson.hasOwnProperty(field.prop)) {
                configJson[field.prop] = '';
              }
            });

            this.form = {
              id: model.id || "",
              modelType: model.modelType || "",
              modelCode: model.modelCode || "",
              modelName: model.modelName || "",
              isDefault: model.isDefault || 0,
              isEnabled: model.isEnabled || 0,
              docLink: model.docLink || "",
              remark: model.remark || "",
              sort: model.sort || 0,
              configJson: {
                ...JSON.parse(JSON.stringify(DEFAULT_CONFIG_JSON)),
                ...configJson,
                config: {
                  ...DEFAULT_CONFIG_JSON.config,
                  ...(configJson.config || {})
                }
              }
            };
          }
        });
      }
    },
    handleSave() {
      const provideCode = this.form.configJson.type;
      const { provider, ...restConfigJson } = this.form.configJson;
      const formData = {
        id: this.form.id,
        modelCode: this.form.modelCode,
        modelName: this.form.modelName,
        isDefault: this.form.isDefault ? 1 : 0,
        isEnabled: this.form.isEnabled ? 1 : 0,
        docLink: this.form.docLink,
        remark: this.form.remark,
        sort: this.form.sort || 0,
        configJson: {
          ...restConfigJson,
          config: {
            ...restConfigJson.config,
            ignoreError: !!restConfigJson.config?.ignoreError,
            ignoreCase: !!restConfigJson.config?.ignoreCase,
          }
        }
      };
      this.$emit("save", { provideCode, formData });
      this.dialogVisible = false;
    },
    loadProviders() {
      if (this.providersLoaded) return;

      Api.model.getModelProviders(this.modelType, (data) => {
        this.providers = data.map(item => ({
          label: item.name,
          value: item.providerCode
        }));
        this.providersLoaded = true;
      });
    },
  }
};
</script>

<style scoped>
.custom-dialog {
  position: relative;
  border-radius: 20px;
  overflow: hidden;
  background: white;
  padding-bottom: 17px;
}

.custom-dialog .el-dialog__header {
  padding: 0;
  border-bottom: none;
}

.center-dialog {
  display: flex;
  align-items: center;
  justify-content: center;
}

.center-dialog .el-dialog {
  margin: 4% 0 auto !important;
  display: flex;
  flex-direction: column;
}

.custom-close-btn {
  position: absolute;
  top: 20px;
  right: 20px;
  width: 35px;
  height: 35px;
  border-radius: 50%;
  border: 2px solid #cfcfcf;
  background: none;
  font-size: 30px;
  font-weight: lighter;
  color: #cfcfcf;
  cursor: pointer;
  display: flex;
  align-items: center;
  justify-content: center;
  z-index: 1;
  padding: 0;
  outline: none;
}

.custom-close-btn:hover {
  color: #409EFF;
  border-color: #409EFF;
}

.custom-select .el-input__suffix {
  background: #e6e8ea;
  right: 6px;
  width: 20px;
  height: 20px;
  display: flex;
  justify-content: center;
  align-items: center;
  top: 9px;
}

.custom-select .el-input__suffix-inner {
  display: flex;
  align-items: center;
  justify-content: center;
  width: 100%;
}

.custom-select .el-icon-arrow-up:before {
  content: "";
  display: inline-block;
  width: 0;
  height: 0;
  border-left: 5px solid transparent;
  border-right: 5px solid transparent;
  border-top: 7px solid #c0c4cc;
  position: relative;
  top: -2px;
  transform: rotate(180deg);
}

.custom-form .el-form-item {
  margin-bottom: 20px;
}

.custom-form .el-form-item__label {
  color: #3d4566;
  font-weight: normal;
  text-align: right;
  padding-right: 20px;
}

.custom-form .el-form-item.prop-remark .el-form-item__label {
  margin-top: -4px;
}

.custom-input-bg .el-input__inner::-webkit-input-placeholder,
.custom-input-bg .el-textarea__inner::-webkit-input-placeholder {
  color: #9c9f9e;
}

.custom-input-bg .el-input__inner,
.custom-input-bg .el-textarea__inner {
  background-color: #f6f8fc;
}

.save-btn {
  background: #e6f0fd;
  color: #237ff4;
  border: 1px solid #b3d1ff;
  width: 150px;
  height: 40px;
  font-size: 16px;
  transition: all 0.3s ease;
}

.save-btn:hover {
  background: linear-gradient(to right, #237ff4, #9c40d5);
  color: white;
  border: none;
}

.custom-switch .el-switch__core {
  border-radius: 20px;
  height: 23px;
  background-color: #c0ccda;
  width: 35px;
  padding: 0 20px;
}

.custom-switch .el-switch__core:after {
  width: 15px;
  height: 15px;
  background-color: white;
  top: 3px;
  left: 4px;
  transition: all .3s;
}

.custom-switch.is-checked .el-switch__core {
  border-color: #b5bcf0;
  background-color: #cfd7fa;
  padding: 0 20px;
}

.custom-switch.is-checked .el-switch__core:after {
  left: 100%;
  margin-left: -18px;
  background-color: #1b47ee;
}

[style*="display: flex"] {
  gap: 20px;
}

.custom-input-bg .el-input__inner {
  height: 32px;
}


.custom-form .el-form-item {
  margin-bottom: 20px;
}


.custom-input-bg .el-input__inner {
  height: 32px;
}


.custom-form .el-form-item__label {
  color: #3d4566;
  font-weight: normal;
  text-align: right;
  padding-right: 20px;
}
</style>
